# Copyright 2008 Brian Boyer, Ryan Mark, Angela Nitzke, Joshua Pollock,
# Stuart Tiffen, Kayla Webley and the Medill School of Journalism, Northwestern
# University.
#
# This file is part of Crunchberry Pie.
#
# Crunchberry Pie is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Crunchberry Pie is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
#You should have received a copy of the GNU General Public License
#along with Crunchberry Pie.  If not, see <http://www.gnu.org/licenses/>.

from django import template
from pie.bumbles.models import Bumble
from pie.letters.models import Letter
from pie.questions.models import Question, Answer

register = template.Library()

@register.inclusion_tag("profiles/activity.html",takes_context=True)
def show_activity(context, activity):
    context.update({
        'bumbles':   activity[Bumble],
        'letters':   activity[Letter],
        'questions': activity[Question],
        'answers':   activity[Answer],
    })
    return context

@register.inclusion_tag("profiles/activity.html",takes_context=True)
def show_article_activity(context, article):
    answers = 0
    for q in article.question_set.all(): answers += q.answer_set.count()
    activity = {
        Bumble:   article.bumble_set.count(),
        Letter:   article.letter_set.count(),
        Question: article.question_set.count(),
        Answer:   answers,
    }
    show_activity(context, activity)
    return context
